# run_train_pymarl.py
import os

# 切换目录到 PyMARL 的 src 目录
os.chdir("pymarl/src")

# 运行训练命令（以 2s3z 地图为例）
os.system("python main.py --config=qmix --env-config=sc2 with env_args.map_name=2s3z_40%")

